package com.ddw.test;

import com.itextpdf.text.Element;
import com.itextpdf.text.pdf.*;

import java.io.FileOutputStream;

/**
 * PDF文件水印添加
 */
public class PDFWaterMarkUtil {

    /**
     * @param srcPath  源文件路径
     * @param destPath 目的文件路径
     * @param word     添加水印（不支持汉字）
     * @throws Exception
     */
    public static void addPDFWaterMark(String srcPath, String destPath, String word)
            throws Exception {

        PdfReader reader = new PdfReader(srcPath);
        PdfStamper stamper = new PdfStamper(reader, new FileOutputStream(destPath));

        //创建字体，第一个参数是字体路径
        BaseFont base = BaseFont.createFont();
        //BaseFont base = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.EMBEDDED);

        PdfGState gs = new PdfGState();
        gs.setFillOpacity(0.2f);//图片水印透明度
        //gs.setStrokeOpacity(0.4f);//设置笔触字体不透明度
        PdfContentByte content = null;

        int total = reader.getNumberOfPages();//pdf文件页数
        for (int i = 0; i < total; i++) {
            float x = reader.getPageSize(i + 1).getWidth();//页宽度
            float y = reader.getPageSize(i + 1).getHeight();//页高度
            content = stamper.getOverContent(i + 1);
            content.setGState(gs);
            content.beginText();//开始写入
            content.setFontAndSize(base, 30);//字体大小
            //每页3行，一行3个
            for (int j = 0; j < 3; j++) {
                for (int k = 0; k < 3; k++) {
                    //showTextAligned 方法的参数（文字对齐方式，位置内容，输出水印X轴位置，Y轴位置，旋转角度）
                    content.showTextAligned(Element.ALIGN_CENTER, word, x / 3 * j + 100, y / 3 * k + 100, 45);
                }
            }
            content.endText();//结束写入
        }
        //关闭流
        stamper.close();
        reader.close();
    }

    public static void main(String[] args) {
        // 获取指定路径的pdf
        try {
            addPDFWaterMark("D:\\20240801-段德文-钱包功能简单介绍加水印版.pdf", "D:\\example_water.pdf", "duandewen@GMM");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

}